'''
角色:姓名，性别，职业
添加角色
删除角色
修改角色
查询角色 单个角色
显示所有角色
退出系统
'''
import time

all_role = [] #存储所有角色
print('---------欢迎进入角色管理系统---------')
while True:
    choice = input('请选择操作：1.添加角色\n2.删除角色\n3.修改角色\n4.查询角色\n5.显示所有角色\n6.退出系统\n')
    if choice == '1':
        name = input('姓名：')
        sex = input('性别：')
        job = input('职业：')
        role = [name,sex,job]
        all_role.append(role)
        print('\t成功添加{}到角色管理系统\n'.format(name))
    elif choice == '2':
        print('删除角色模块：')
        role_name = input('请输入要删除的角色姓名：')
        for role in all_role:
            if role[0] == role_name:
                print('是否确定删除{}角色(y/n)'.format(role_name))
                a = input()
                if a == 'y':
                    all_role.remove(role)
                    print('成功删除{}角色'.format(role_name))
                    break
                else:
                    print('取消删除')
        else:
            print('没有找到{}角色'.format(role_name))
    elif choice == '3':
        print('修改角色模块：')
        role_name = input('请输入要修改的角色姓名：')
        for role in all_role:
            if role[0] == role_name:
                role[1] = input('请输入修改后的性别：')
                role[2] = input('请输入修改后的职业：')
                print('成功修改{}角色信息'.format(role_name))
                break
        else:
            print('没有找到{}角色'.format(role_name))
    elif choice == '4':
        print('查询角色模块：')
        role_name = input('\t姓名：')
        for role in all_role:
            if role[0] == role_name:
                print('\t存在此角色，信息如下：')
                print('\t{}{}{}'.format(role[0].center(10),role[1].center(10),role[2].center(10)))
                break
            else:
                print('\t不存在此角色{},请检查角色名称'.format(role_name))
    elif choice == '5':
        print('显示所有角色模块：')
        print('{}{}{}'.format('姓名'.center(10),'性别'.center(10),'职业'.center(10)))
        for role in all_role:
            print(role[0].center(10),end='')
            print(role[1].center(10),end='')
            print(role[2].center(10),end='')
            print()
    elif choice == '6':
        print('感谢使用角色管理系统，再见！')
        time.sleep(3)
        print('成功退出！')
        break
    else:
        print('输入错误，请重新输入！')
